#region Copyright (c) 2006-2012 nHydrate.org, All Rights Reserved
// --------------------------------------------------------------------- *
//                           NHYDRATE.ORG                                *
//              Copyright (c) 2006-2012 All Rights reserved              *
//                                                                       *
//                                                                       *
// This file and its contents are protected by United States and         *
// International copyright laws.  Unauthorized reproduction and/or       *
// distribution of all or any portion of the code contained herein       *
// is strictly prohibited and will result in severe civil and criminal   *
// penalties.  Any violations of this copyright will be prosecuted       *
// to the fullest extent possible under law.                             *
//                                                                       *
// THE SOURCE CODE CONTAINED HEREIN AND IN RELATED FILES IS PROVIDED     *
// TO THE REGISTERED DEVELOPER FOR THE PURPOSES OF EDUCATION AND         *
// TROUBLESHOOTING. UNDER NO CIRCUMSTANCES MAY ANY PORTION OF THE SOURCE *
// CODE BE DISTRIBUTED, DISCLOSED OR OTHERWISE MADE AVAILABLE TO ANY     *
// THIRD PARTY WITHOUT THE EXPRESS WRITTEN CONSENT OF THE NHYDRATE GROUP *
//                                                                       *
// UNDER NO CIRCUMSTANCES MAY THE SOURCE CODE BE USED IN WHOLE OR IN     *
// PART, AS THE BASIS FOR CREATING A PRODUCT THAT PROVIDES THE SAME, OR  *
// SUBSTANTIALLY THE SAME, FUNCTIONALITY AS THIS PRODUCT                 *
//                                                                       *
// THE REGISTERED DEVELOPER ACKNOWLEDGES THAT THIS SOURCE CODE           *
// CONTAINS VALUABLE AND PROPRIETARY TRADE SECRETS OF NHYDRATE,          *
// THE REGISTERED DEVELOPER AGREES TO EXPEND EVERY EFFORT TO             *
// INSURE ITS CONFIDENTIALITY.                                           *
//                                                                       *
// THE END USER LICENSE AGREEMENT (EULA) ACCOMPANYING THE PRODUCT        *
// PERMITS THE REGISTERED DEVELOPER TO REDISTRIBUTE THE PRODUCT IN       *
// EXECUTABLE FORM ONLY IN SUPPORT OF APPLICATIONS WRITTEN USING         *
// THE PRODUCT.  IT DOES NOT PROVIDE ANY RIGHTS REGARDING THE            *
// SOURCE CODE CONTAINED HEREIN.                                         *
//                                                                       *
// THIS COPYRIGHT NOTICE MAY NOT BE REMOVED FROM THIS FILE.              *
// --------------------------------------------------------------------- *
#endregion
using System;
using System.Text;
using nHydrate.Generator.Models;

namespace nHydrate.Generator.EFDAL.Mocks.Generators.MockObjectSets
{
	public class MockObjectSetGeneratedTemplate : EFDALMockBaseTemplate
	{
		private readonly StringBuilder sb = new StringBuilder();

		public MockObjectSetGeneratedTemplate(ModelRoot model)
			: base(model)
		{
		}

		#region BaseClassTemplate overrides
		public override string FileName
		{
			get { return "MockObjectSet.Generated.cs"; }
		}

		public string ParentItemName
		{
			get { return "MockObjectSet.cs"; }
		}

		public override string FileContent
		{
			get
			{
				GenerateContent();
				return sb.ToString();
			}
		}
		#endregion

		#region GenerateContent

		private void GenerateContent()
		{
			try
			{
				nHydrate.Generator.GenerationHelper.AppendFileGeneatedMessageInCode(sb);
				nHydrate.Generator.GenerationHelper.AppendCopyrightInCode(sb, _model);
				this.AppendUsingStatements();
				sb.AppendLine("namespace " + this.GetLocalNamespace());
				sb.AppendLine("{");
				sb.AppendLine("	/// <summary>");
				sb.AppendLine("	/// This object is used to mock entity sets on the mock context");
				sb.AppendLine("	/// </summary>");
				sb.AppendLine("	public partial class MockObjectSet<T> : IObjectSet<T> where T : class");
				sb.AppendLine("	{");
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// ");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		protected HashSet<T> _data;");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// ");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		protected IQueryable _query;");
				sb.AppendLine();
				sb.AppendLine("		#region Constructor");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// Default constructor for MockObjectSet");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public MockObjectSet() : this(new List<T>()) { }");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// ");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public MockObjectSet(IEnumerable<T> initialData)");
				sb.AppendLine("		{");
				sb.AppendLine("			_data = new HashSet<T>(initialData);");
				sb.AppendLine("			_query = _data.AsQueryable();");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		#endregion");
				sb.AppendLine();
				sb.AppendLine("		#region Methods");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// Adds an item to this list");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public void Add(T item)");
				sb.AppendLine("		{");
				sb.AppendLine("			_data.Add(item);");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// Adds an item to this list");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public void AddObject(T item)");
				sb.AppendLine("		{");
				sb.AppendLine("			_data.Add(item);");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// Removes an item to this list");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public void Remove(T item)");
				sb.AppendLine("		{");
				sb.AppendLine("			_data.Remove(item);");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// Removes an item to this list");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public void DeleteObject(T item)");
				sb.AppendLine("		{");
				sb.AppendLine("			_data.Remove(item);");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// Attaches an item to this list");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public void Attach(T item)");
				sb.AppendLine("		{");
				sb.AppendLine("			_data.Add(item);");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		/// <summary>");
				sb.AppendLine("		/// Detaches an item to this list");
				sb.AppendLine("		/// </summary>");
				sb.AppendLine("		public void Detach(T item)");
				sb.AppendLine("		{");
				sb.AppendLine("			_data.Remove(item);");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		#endregion");
				sb.AppendLine();
				sb.AppendLine("		#region Base Functionality");
				sb.AppendLine();
				sb.AppendLine("		Type IQueryable.ElementType");
				sb.AppendLine("		{");
				sb.AppendLine("			get { return _query.ElementType; }");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		System.Linq.Expressions.Expression IQueryable.Expression");
				sb.AppendLine("		{");
				sb.AppendLine("			get { return _query.Expression; }");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		IQueryProvider IQueryable.Provider");
				sb.AppendLine("		{");
				sb.AppendLine("			get { return _query.Provider; }");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()");
				sb.AppendLine("		{");
				sb.AppendLine("			return _data.GetEnumerator();");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		IEnumerator<T> IEnumerable<T>.GetEnumerator()");
				sb.AppendLine("		{");
				sb.AppendLine("			return _data.GetEnumerator();");
				sb.AppendLine("		}");
				sb.AppendLine();
				sb.AppendLine("		#endregion");
				sb.AppendLine();
				sb.AppendLine("	}");
				sb.AppendLine();
				sb.AppendLine("}");
				sb.AppendLine();
			}
			catch (Exception ex)
			{
				throw;
			}
		}

		private void AppendUsingStatements()
		{
			sb.AppendLine("using System;");
			sb.AppendLine("using System.Linq;");
			sb.AppendLine("using System.Data.Objects;");
			sb.AppendLine("using System.Data.Objects.DataClasses;");
			sb.AppendLine("using System.ComponentModel;");
			sb.AppendLine("using System.Runtime.Serialization;");
			sb.AppendLine("using System.Collections.Generic;");
			sb.AppendLine("using " + this.GetLocalNamespace() + ".Entity;");
			sb.AppendLine();
		}

		#endregion

	}
}